how to create python function